package org.tity.commons.network;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;

import org.tity.commons.log.Logger;

public class HttpUtil {

	public static String downLoadFromUrl(String urlStr, File f) {

		if (f.exists()) {
			f.delete();
		}

		f.getParentFile().mkdirs();

		String savePath = f.getParent();
		String fileName = f.getName();
		FileOutputStream fos=null;
		try {

			URL url = new URL(urlStr);
			HttpURLConnection conn = (HttpURLConnection) url.openConnection();
			// 设置超时间为3秒
			conn.setConnectTimeout(3 * 1000);
			// 防止屏蔽程序抓取而返回403错误
			conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");

			// 得到输入流
			InputStream inputStream = conn.getInputStream();
			// 获取自己数组
			byte[] getData = readInputStream(inputStream);

			// 文件保存位置
			File saveDir = new File(savePath);
			if (!saveDir.exists()) {
				saveDir.mkdir();
			}
			File file = new File(saveDir + File.separator + fileName);
			fos = new FileOutputStream(file);
			fos.write(getData);
			fos.close();
			inputStream.close();
			return saveDir + File.separator + fileName;
		} catch (Exception e) {
			Logger.error(e);
		} finally {
			if(fos!=null) {
				try {
					fos.close();
				} catch (IOException e) {
					 Logger.error(e);
				}
			}
		}
		return "";

	}

	public static byte[] readInputStream(InputStream inputStream) throws IOException {
		byte[] buffer = new byte[1024];
		int len = 0;
		ByteArrayOutputStream bos = new ByteArrayOutputStream();
		while ((len = inputStream.read(buffer)) != -1) {
			bos.write(buffer, 0, len);
		}
		bos.close();
		return bos.toByteArray();
	}

}
